package com.atguigu.gulimall.order.config;


import lombok.extern.slf4j.Slf4j;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import javax.annotation.PostConstruct;

@Slf4j
@Configuration
public class MyRabbitConfig {


    @Autowired
    RabbitTemplate rabbitTemplate;


    private final Logger logger = LoggerFactory.getLogger(MyRabbitConfig.class);

    @Bean
    public RabbitTemplate createRabbitTemplate(ConnectionFactory connectionFactory){
        RabbitTemplate rabbitTemplate = new RabbitTemplate();
        rabbitTemplate.setConnectionFactory(connectionFactory);
        //设置开启Mandatory,才能触发回调函数,无论消息推送结果怎么样都强制调用回调函数
        rabbitTemplate.setMandatory(true);
        rabbitTemplate.setMessageConverter(new Jackson2JsonMessageConverter());
        rabbitTemplate.setConfirmCallback((correlationData, ack, cause) -> {
            //成功发送到交换机会触发该回调
            if (ack){
                //表示消息发送成功，这里不做任何操作，这里只是为了写个注释（手动狗头）
            }else {
                //表示消息成功发送到服务器，但是没有找到交换器，这里可以记录日志，方便后续处理
                logger.warn("ConfirmCallback -> 消息发布到交换器失败，错误原因为：{}", cause);
            }
        });

        rabbitTemplate.setReturnCallback((message, replyCode, replyText, exchange, routingKey) -> {
            //表示消息发送到交换机，但是没有找到队列，这里记录日志
            logger.warn("ReturnCallback -> 消息{}，发送到队列失败，应答码：{}，原因：{}，交换器: {}，路由键：{}",
                    message,
                    replyCode,
                    replyText,
                    exchange,
                    routingKey);
        });

        return rabbitTemplate;
    }


    @Primary
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        rabbitTemplate.setMandatory(true);

        this.rabbitTemplate = rabbitTemplate;

        rabbitTemplate.setMessageConverter(messageConverter());

        initRabbitTemplate();
        return rabbitTemplate;
    }


    /**
     * 定义RabbitMQ序列化器
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     * 1、服务收到消息回调
     *      1 spring.rabbitmq.publisher-confirm-type=correlated
     *      2 设置确认回调 ConfirmCallback
     * 2、消息正确抵达队列进行回调
     *      1 spring.rabbitmq.publisher-returns=true
     *        spring.rabbitmq.template.mandatory=true
     *      2 设置确认回调ReturnCallback
     * 3、消费端确认（保证每个消息被正确的消费，此时才可以broker删除这个消息）
     *      spring.rabbitmq.listener.simple.acknowledge-mode=manual
     *      1 默认是自动确认的，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     *          问题：收到多个消息，只处理了一个之后就宕机，消息会发生丢失
     *          开启手动确认模式，只要我们没明确告诉MQ，没有ack，消息一直就是unacked，即使宕机，也不会丢失。
     *      2 如何签收
     *        channel.basicAck(deliveryTag,false) 签收 业务成功完成
     *        channel.basicNAck(deliveryTag,false,true) 拒签并投递回去
     */


//    @PostConstruct // MyRabbitConfig对象创建完成以后，调用这个方法
    public void initRabbitTemplate() {

        // 设置回调 消息抵达Broker回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 只要消息抵达Broker ack就会是true
             * @param correlationData 当前消息唯一关联数据（唯一id）
             * @param ack 消息是否成功收到
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                // 代表服务器收到了
                /*
                    防止消息丢失
                        1、做好消息确认机制（publisher，consumer【手动ack】）
                        2、每一个发送的消息都在数据库做好记录，定期将失败的消息再次发送一遍
                 */
//                System.out.println("confirm...correlationData[{}],ack[{}],cause[{}]"+"-----"+correlationData+"-----"+ ack+"-----"+ cause);
            }
        });


        // 设置消息抵达队列Queue的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message 投递失败的消息的详细信息
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange 发送时的交换机
             * @param routingKey 发送时的路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // 报错了，修改数据库当前消息的状态-》错误
                System.out.println("Fail Message[{}],replyCode[{}]"+"-----"+message+"-----"+replyCode);
            }
        });




    }
}